
package org.springblade.modules.auth.granter;

import lombok.extern.slf4j.Slf4j;
import org.springblade.common.cache.CacheNames;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.redis.cache.BladeRedis;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.core.tool.utils.WebUtil;
import org.springblade.modules.auth.enums.BladeUserEnum;
import org.springblade.modules.auth.provider.ITokenGranter;
import org.springblade.modules.auth.provider.TokenParameter;
import org.springblade.modules.auth.utils.TokenUtil;
import org.springblade.modules.system.entity.Tenant;
import org.springblade.modules.system.entity.UserInfo;
import org.springblade.modules.system.service.ITenantService;
import org.springblade.modules.system.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**
 * 验证码TokenGranter
 *
 * @author Chill
 */
@Slf4j
@Component
public class CaptchaTokenGranter implements ITokenGranter {

	public static final String GRANT_TYPE = "captcha";

	@Autowired
	private IUserService userService;
	@Autowired
	private ITenantService tenantService;
	@Autowired
	private BladeRedis bladeRedis;

	@Override
	public UserInfo grant(TokenParameter tokenParameter) {
		HttpServletRequest request = WebUtil.getRequest();

		String tenantId = tokenParameter.getArgs().getStr("tenantId");
		String phone = tokenParameter.getArgs().getStr("phone");
		String smscode = tokenParameter.getArgs().getStr("smscode");

//		String key = request.getHeader(TokenUtil.CAPTCHA_HEADER_KEY);
//		String code = request.getHeader(TokenUtil.CAPTCHA_HEADER_CODE);
		// 获取验证码
		String redisCode = bladeRedis.get(CacheNames.CAPTCHA_KEY_SMS + phone);
		// 判断验证码
		if (smscode == null || !StringUtil.equalsIgnoreCase(redisCode, smscode)) {
			throw new ServiceException(TokenUtil.CAPTCHA_NOT_CORRECT);
		}


		UserInfo userInfo = null;
		// 获取租户信息
		Tenant tenant = tenantService.getByTenantId(tenantId);
		if (TokenUtil.judgeTenant(tenant)) {
			throw new ServiceException(TokenUtil.USER_HAS_NO_TENANT_PERMISSION);
		}
		// 获取用户类型
		String userType = tokenParameter.getArgs().getStr("userType");
		// 根据不同用户类型调用对应的接口返回数据，用户可自行拓展
		if (userType.equals(BladeUserEnum.WEB.getName())) {
			userInfo = userService.userInfoByPhone(tenantId, phone);
		} else if (userType.equals(BladeUserEnum.APP.getName())) {
			userInfo = userService.userInfoByPhone(tenantId, phone);
		} else {
			userInfo = userService.userInfoByPhone(tenantId, phone);
		}
		return userInfo;
	}

}
